package com.ruoyi.gqgl.rdb.service.impl;

import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.gqgl.rdb.domain.PSRDBhCk;
import com.ruoyi.gqgl.rdb.domain.Rdb;
import com.ruoyi.gqgl.rdb.domain.RdbFj;
import com.ruoyi.gqgl.rdb.mapper.RdbMapper;
import com.ruoyi.gqgl.rdb.service.IRdbService;
import com.ruoyi.gqgl.rdjh.domain.Rdjh;
import com.ruoyi.gqgl.rdjh.mapper.RdjhMapper;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.validation.annotation.Validated;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

import static com.ruoyi.common.utils.SecurityUtils.*;

/**
 * RDService业务层处理
 *
 * @author huacai
 * @date 2025-01-23
 */
@Service
public class RdbServiceImpl implements IRdbService {
    @Resource
    private RdbMapper rdbMapper;

    @Resource
    private SqlSessionFactory sqlSessionFactory;

    @Resource
    private RdjhMapper rdjhMapper;

    /**
     * 查询RD
     *
     * @param rdbId RD主键
     * @return RD
     */
    @Override
    public Rdb selectRdbByRdbId(String rdbId) {
        return rdbMapper.selectRdbByRdbId(rdbId);
    }

    /**
     * 查询RD列表
     *
     * @param rdb RD
     * @return RD
     */
    @Override
    public List<Rdb> selectRdbList(Rdb rdb) {
        return rdbMapper.selectRdbList(rdb);
    }

    /**
     * 新增RD
     *
     * @param rdb RD
     * @return 结果
     */
    @Transactional
    @Override
    public int insertRdb(@Validated Rdb rdb) {
        rdb.setUserId(Math.toIntExact(getUserId())); //用户ID
        rdb.setDeptId(Math.toIntExact(getDeptId())); //部门ID
        rdb.setCreateBy(getUsername()); //创建人
        rdb.setIdWhere("存在");
        rdb.setCreateTime(DateUtils.getNowDate());

        // 检查当前年度是否有已存在的记录，以确定RD编号
        Integer slByRdNd = rdbMapper.selectRdbByrdNd(rdb.getNd());
        if (slByRdNd == null) {
            // 如果没有找到记录，RD编号从"RD001"开始
            rdb.setRdBh("RD001");
        } else {
            // 如果找到记录，RD编号在原有基础上递增
            rdb.setRdBh("RD" + String.format("%03d", slByRdNd + 1));
        }

        Rdjh rdjh = new Rdjh();
        rdjh.setRdjhId(rdb.getRdjhId());
        rdjh.setSfzSqLb("是");
        rdjhMapper.updateRdjh(rdjh);

        int rows = rdbMapper.insertRdb(rdb);
        insertRdbFj(rdb);
        return rows;
    }

    /**
     * 批量新增RD
     *
     * @param rdbs RDList
     * @return 结果
     */
    @Override
    public int batchInsertRdb(List<Rdb> rdbs) {
        SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false);
        int count = 0;
        if (!CollectionUtils.isEmpty(rdbs)) {
            try {
                for (int i = 0; i < rdbs.size(); i++) {
                    int row = rdbMapper.insertRdb(rdbs.get(i));
                    // 防止内存溢出，每100次提交一次,并清除缓存
                    boolean bool = (i > 0 && i % 100 == 0) || i == rdbs.size() - 1;
                    if (bool) {
                        sqlSession.commit();
                        sqlSession.clearCache();
                    }
                    count = i + 1;
                }
            } catch (Exception e) {
                e.printStackTrace();
                // 没有提交的数据可以回滚
                sqlSession.rollback();
            } finally {
                sqlSession.close();
                return count;
            }
        }
        return count;
    }

    /**
     * 修改RD
     *
     * @param rdb RD
     * @return 结果
     */
    @Transactional
    @Override
    public int updateRdb(Rdb rdb) {
        rdb.setUpdateTime(DateUtils.getNowDate());
//        rdbMapper.deleteRdbFjByRdbId(rdb.getRdbId());
        insertRdbFj(rdb);
        return rdbMapper.updateRdb(rdb);
    }

    /**
     * 批量删除RD
     *
     * @param rdbIds 需要删除的RD主键
     * @return 结果
     */
    @Transactional
    @Override
    public int deleteRdbByRdbIds(String[] rdbIds) {
        for (String rdbId : rdbIds) {
            Rdb rdb = rdbMapper.selectRdbByRdbId(rdbId);

            Rdjh rdjh = new Rdjh();
            rdjh.setRdjhId(rdb.getRdjhId());
            rdjh.setSfzSqLb("否");
            rdjhMapper.updateRdjh(rdjh);
        }
        rdbMapper.deleteRdbFjByRdbIds(rdbIds);
        return rdbMapper.deleteRdbByRdbIds(rdbIds);
    }

    /**
     * 删除RD信息
     *
     * @param rdbId RD主键
     * @return 结果
     */
    @Transactional
    @Override
    public int deleteRdbByRdbId(String rdbId) {
        rdbMapper.deleteRdbFjByRdbId(rdbId);
        return rdbMapper.deleteRdbByRdbId(rdbId);
    }

    /**
     * 新增附件
     * @param rdbFj
     * @return
     */
    @Override
    public int insertRdbFjByRdbId(RdbFj rdbFj) {
        rdbFj.setCreateBy(getUsername());
        rdbFj.setDeptId(Math.toIntExact(getDeptId()));
        rdbFj.setUserId(Math.toIntExact(getUserId()));
        return rdbMapper.insertRdbFjByRdbId(rdbFj);
    }

    /**
     * 根据ID查询附件列表
     * @param rdbFj
     * @return
     */
    @Override
    public List<RdbFj> selectRdbFjList(RdbFj rdbFj) {
        return rdbMapper.selectRdbFjList(rdbFj);
    }

    /**
     * 删除RB附件
     * @param fjIds
     * @return
     */
    @Override
    public int deleteRdbFjByFjIds(Integer[] fjIds) {
        return rdbMapper.deleteRdbFjByFjIds(fjIds);
    }

    /**
     * PS/RD编号查看
     * @param nd
     * @return
     */
    @Override
    public List<PSRDBhCk> selectPSRDBh(PSRDBhCk psrdBhCk) {
        return rdbMapper.selectPSRDBh(psrdBhCk);
    }

    /**
     * 新增RB附件信息
     *
     * @param rdb RD对象
     */
    public void insertRdbFj(Rdb rdb) {
        List<RdbFj> rdbFjList = rdb.getRdbFjList();
        String rdbId = rdb.getRdbId();
        if (StringUtils.isNotNull(rdbFjList)) {
            List<RdbFj> list = new ArrayList<RdbFj>();
            for (RdbFj rdbFj : rdbFjList) {
                rdbFj.setRdbId(rdbId);
                list.add(rdbFj);
            }
//            if (list.size() > 0) {
//                rdbMapper.batchRdbFj(list);
//            }
        }
    }
}
